# -*- coding: utf-8 -*-
# @Time    : 2021/10/17 10:47
# @Author  : super

'''
需求：批量修改⽂件名，既可添加指定字符串，⼜能删除指定字符串。
步骤
1. 设置添加删除字符串的的标识
2. 获取指定⽬录的所有⽂件
3. 将原有⽂件名添加/删除指定字符串，构造新名字
4. os.rename()重命名
'''
import os

flag = int(input("选择您要修改字符串还是删除字符串(1.添加，2.删除)"))
str_name = input("输入字符串：")
if flag == 1:
	for item in os.listdir():
		os.rename(item,str_name+item)
elif flag == 2:
	for item in os.listdir():
		num = len(str_name)
		new_name = item[num:]
		os.rename(item,new_name)
else:
	pass
